import json

def filter_json_data():
    try:
        # 硬编码的输入和输出文件路径
        input_file = '/Users/liuwei/Library/Mobile Documents/com~apple~CloudDocs/混杂任务执行/simple_tasks.json'
        output_file = 'video_tasks.json'
        
        # 读取输入的 JSON 文件
        with open(input_file, 'r', encoding='utf-8') as f:
            data = json.load(f)
        
        # 创建一个新的字典，用于存储过滤后的结果
        filtered_data = {}
        
        # 遍历原始数据中的每个项目
        for key, item in data.items():
            # 检查 modality 字段是否存在且不包含 'v'
            if 'modality' in item and 'v' in item['modality']:
                filtered_data[key] = item
        
        # 将过滤后的数据写入输出文件
        with open(output_file, 'w', encoding='utf-8') as f:
            json.dump(filtered_data, f, ensure_ascii=False, indent=4)
        
        print(f"过滤完成，已将结果保存到 {output_file}")
    
    except FileNotFoundError:
        print(f"错误：找不到输入文件 {input_file}")
    except json.JSONDecodeError:
        print(f"错误：输入文件 {input_file} 不是有效的 JSON 格式")
    except Exception as e:
        print(f"发生未知错误：{e}")

if __name__ == "__main__":
    # 执行过滤操作
    filter_json_data()    